package 简单.双指针;

/**
 * 给定一个已按照 升序排列  的整数数组 numbers ，请你从数组中找出两个数满足相加之和等于目标数 target 。
 * <p>
 * 函数应该以长度为 2 的整数数组的形式返回这两个数的下标值。numbers 的下标 从 1 开始计数 ，所以答案
 * 数组应当满足 1 <= answer[0] < answer[1] <= numbers.length 。
 * <p>
 * 你可以假设每个输入只对应唯一的答案，而且你不可以重复使用相同的元素。
 * <p>
 * 来源：https://leetcode-cn.com/problems/two-sum-ii-input-array-is-sorted/
 */
public class 两数之和II输入有序数组_167 {

    public static void main(String[] args) {
        int[] nums = {5, 25, 75};
        int[] ints = twoSum2(nums, 100);
        System.out.println(ints[0]);
        System.out.println(ints[1]);
    }

    //暴力解法，针对无序数组的，没有利用到输入数组有序的性质
    public static int[] twoSum(int[] numbers, int target) {
        int[] res = new int[2];
        for (int i = 0; i < numbers.length; i++) {
            int j = i + 1;
            while (j < numbers.length) {
                if (target == numbers[i] + numbers[j]) {
                    res[0] = i + 1;
                    res[1] = j + 1;
                    return res;
                }
                j++;
            }
        }
        return res;
    }

    //二分法，先固定一个数，另一个数使用二分法进行查找
    //如果target-固定的数=另一个数，找到
    //如果target-固定的数<另一个数，另一个数向后移一位
    //如果target-固定的数>另一个数，另一个数向前移一位
    public static int[] twoSum2(int[] numbers, int target) {
        int[] res = new int[2];
        for (int i = 0; i < numbers.length; i++) {
            int two = target - numbers[i];
            int left = i + 1;
            int right = numbers.length - 1;

            while (left <= right) {
                int mid = left + (right - left) / 2;
                if (numbers[mid] == two) {
                    res[0] = i + 1;
                    res[1] = mid + 1;
                    return res;
                } else if (numbers[mid] < two) {
                    left = mid + 1;
                } else {
                    right = mid - 1;
                }
            }
        }
        return res;
    }

    //双指针
    //一个指针指向首部，一个指针指向尾部
    //如果两指针相加大于目标值，则尾部指针向前移一位
    //如果两指针相加小于目标值，则首部指针向后移一位
    public static int[] efficientTwoSum(int[] numbers, int target) {
        int[] res = new int[2];
        for (int i = 0, j = numbers.length - 1; i < numbers.length; ) {
            if (numbers[i] + numbers[j] == target) {
                res[0] = i + 1;
                res[1] = j + 1;
                return res;
            } else if (numbers[i] + numbers[j] > target) {
                j--;
            } else {
                i++;
            }
        }
        return res;
    }

    /**
     * 双指针
     */
    public int[] twoSum3(int[] numbers, int target) {
        int left = 0, right = numbers.length - 1;
        while (left < right) {
            while (left < right && numbers[left] + numbers[right] > target) {
                right--;
            }
            if (left < right && numbers[left] + numbers[right] == target) {
                return new int[]{left + 1, right + 1};
            }
            left++;
        }
        return new int[]{};
    }
}
